"""
 使用文件和目录操作，定义一个统计指定目录大小的函数（注意目录中还有子
目录）
"""
import os


def get_dir_capacity(dir1):
    """

    :param dir1: 目标文件夹
    :return:
    """
    # 统计文件夹大小
    file_size = 0

    # 获取目录信息
    f_list = os.listdir(dir1)

    # 遍历目录信息
    for f in f_list:
        # 获取文件路径
        file1 = os.path.join(dir1, f)

        # 如果是文件，计算大小
        if os.path.isfile(file1):
            print('文件大小：', os.path.getsize(file1))
            file_size = file_size + os.path.getsize(file1)
            print('当前已统计文件大小：', file_size)
        # 如果是文件夹，遍历
        if os.path.isdir(file1):
            file_size = file_size + get_dir_capacity(file1)
    return file_size


# 测试
print('文件夹大小为：', get_dir_capacity('./test'))
